package net.xdclass.config;

import org.springframework.ai.chat.client.ChatClient;
import org.springframework.ai.chat.client.advisor.MessageChatMemoryAdvisor;
import org.springframework.ai.chat.client.advisor.SimpleLoggerAdvisor;
import org.springframework.ai.chat.memory.ChatMemory;
import org.springframework.ai.openai.OpenAiChatModel;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
public class Config {

//    @Bean
//    ChatClient chatClient(ChatClient.Builder chatClientBuilder) {
//        ChatClient chatClient = chatClientBuilder.defaultSystem("你是一个热爱阅读文学作品的反派，因为你长期遭受生活和工作的压迫，你看清了" +
//                "这个世界的世态炎凉，你每天就是通过阅读古今中外的文学名著来获得心理上的慰藉，你可以回答任何关于文学作品的问题，语言幽默风趣，直白不绕弯子，" +
//                "能够说到问题的核心，不怕丢脸，不会为了面子说假话")
//                .defaultAdvisors(new SimpleLoggerAdvisor()).build();
//
//        return chatClient;
//    }

    @Bean
    ChatClient chatClient(OpenAiChatModel model, ChatMemory chatMemory) {
        ChatClient chatClient = ChatClient.builder(model)
                .defaultAdvisors(MessageChatMemoryAdvisor.builder(chatMemory).build(), new SimpleLoggerAdvisor())
                .defaultSystem("你是小滴课堂超级智能助手 老王，" +
                        "可以回答IT任何技术问题，要幽默风趣的形式")
                .build();

        return chatClient;
    }
}
